

# Hyperparameter
device = 'cuda'
batch_size = 3
lr = 1e-4
epochs = 2
temp = 0.1
num_folds = 2

# Path
# Abs path in Jianing computer
# brain_root = r"F:\ADNI\ADNI_PNG_3Ddata\download_data\NIFTI_data\NIFIT3\all"
# hippocampus_root = r"F:\ADNI\ADNI_PNG_3Ddata\download_data\NIFTI_data\hippdata\all"

# Abs path in 4060Ti
brain_root = r"I:\Nico\AD_program\NIFTI_data\NIFIT3"
hippocampus_root = r"I:\Nico\AD_program\NIFTI_data\hippdata"